package week_07;

import java.util.HashSet;

/**
 * 547. 省份数量
 * 有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，
 * 且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。
 * <p>
 * 省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。
 * <p>
 * 给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，
 * 而 isConnected[i][j] = 0 表示二者不直接相连。
 * <p>
 * 返回矩阵中 省份 的数量。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * <p>
 * 输入：isConnected = [[1,1,0],[1,1,0],[0,0,1]]
 * 输出：2
 * 示例 2：
 * <p>
 * <p>
 * 输入：isConnected = [[1,0,0],[0,1,0],[0,0,1]]
 * 输出：3
 */
public class FindCircleNum_547 {
    int[] fa;

    public int findCircleNum(int[][] isConnected) {
        int n = isConnected.length;
        fa = new int[n];
        for (int i = 0; i < n; i++) fa[i] = i;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (isConnected[i][j] == 1) {
                    unionSet(i, j);
                }
            }
        }
        // 用这种hashset 的方式没法争取的判断 有几组fa
//        HashSet<Integer> set=new HashSet<>();
//        for(int item:fa){
//            set.add(item);
//        }
//        return set.size();
        int ans=0;
        // 当且仅当 fa[i]=i 时为一个新组
        for(int i=0;i<n;i++) if(i==fa[i]) ans++;
        return ans;
    }

    public int find(int x) {
        if (x == fa[x]) return x;
        return fa[x] = find(fa[x]);
    }

    public void unionSet(int x, int y) {
        x = find(x);
        y = find(y);
       if(x!=y) fa[x] = y;
    }

    public static void main(String[] args) {
        FindCircleNum_547 test=new FindCircleNum_547();
        int[][] isConnected =new int[][]{{1,0,0,1}, {0,1,1,0},{0,1,1,1},{1,0,1,1}};
        test.findCircleNum(isConnected);
    }
}
